import os
from dotenv import load_dotenv

load_dotenv()

class Config:
    # DeepSeek API配置
    DEEPSEEK_API_KEY = os.getenv('DEEPSEEK_API_KEY', 'your-deepseek-api-key-here')
    DEEPSEEK_API_BASE = "https://api.deepseek.com/v1"
    
    # 系统配置
    MAX_RETRIES = 3
    REQUEST_TIMEOUT = 60
    BATCH_SIZE = 5
    
    # 题型配置
    QUESTION_TYPES = {
        "single_choice": "单选题",
        "multiple_choice": "多选题", 
        "matrix_scale": "矩阵量表题",
        "open_ended": "开放题"
    }
    
    # 默认智能体配置（备用）
    AGENTS = [
        {
            "name": "大学生-理工科",
            "background": "22岁计算机专业大学生，理性思维强，喜欢数据分析",
            "personality": "严谨、逻辑性强、注重细节"
        },
        {
            "name": "职场白领-市场部",
            "background": "30岁市场专员，有5年工作经验，擅长沟通",
            "personality": "外向、有创意、善于表达"
        },
        {
            "name": "自由职业者-设计师",
            "background": "28岁平面设计师，工作灵活，追求生活品质",
            "personality": "感性、有艺术细胞、独立思考"
        },
        {
            "name": "中年管理者",
            "background": "45岁部门经理，管理经验丰富，注重效率",
            "personality": "稳重、务实、结果导向"
        },
        {
            "name": "退休教师",
            "background": "65岁退休语文教师，教学经验丰富",
            "personality": "耐心、细致、喜欢分享"
        },
        {
            "name": "创业者",
            "background": "35岁科技公司创始人，有冒险精神",
            "personality": "积极、果断、有远见"
        },
        {
            "name": "医护人员",
            "background": "32岁护士，工作认真负责",
            "personality": "细心、有同理心、专业"
        },
        {
            "name": "公务员",
            "background": "38岁政府工作人员，做事规范",
            "personality": "谨慎、守规矩、服务意识强"
        }
    ]